package com.lilei.bigdata.mr.flowsum;


import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.Path;
import org.apache.hadoop.io.LongWritable;
import org.apache.hadoop.io.NullWritable;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Job;
import org.apache.hadoop.mapreduce.Mapper;
import org.apache.hadoop.mapreduce.Reducer;
import org.apache.hadoop.mapreduce.lib.input.FileInputFormat;
import org.apache.hadoop.mapreduce.lib.output.FileOutputFormat;

import java.io.IOException;

public class FlowCountSort {

    static class FlowCountSortMapper extends Mapper<LongWritable, Text,FlowBean,Text>{
        FlowBean bean =new FlowBean();
        Text v = new Text();
        @Override
        protected void map(LongWritable key, Text value, Context context) throws IOException, InterruptedException {
            String line = value.toString();

            String[] fields = line.split("\t");
            String phoneNum = fields[0];
            long upflow = Long.parseLong(fields[1]);
            long dflow = Long.parseLong(fields[2]);
            bean.set(upflow,dflow);
            v.set(phoneNum);
            context.write(bean,v);
        }
    }

    static class FlowCountSortReducer extends Reducer<FlowBean,Text,Text, FlowBean>{
        @Override
        protected void reduce(FlowBean flowbean, Iterable<Text> values, Context context) throws IOException, InterruptedException {
            context.write(values.iterator().next(),flowbean);
        }
    }


    public static void main(String[] args) throws Exception{
        Configuration configuration = new Configuration();
        Job job =   Job.getInstance(configuration);


        //指定本程序的jar包所在的本地路径
        job.setJarByClass(FlowCountSort.class);

        //指定本业务job要使用的mapper/Reducer业务类
        job.setMapperClass(FlowCountSortMapper.class);
        job.setReducerClass(FlowCountSortReducer.class);

        //指定mapper输出数据的kv类型
        job.setMapOutputKeyClass(FlowBean.class);
        job.setMapOutputValueClass(Text.class);

        //指定最终输出的kv类型
        job.setOutputKeyClass(Text.class);
        job.setOutputValueClass(FlowBean.class);

        //指定job的输入原始文件所在目录
        FileInputFormat.setInputPaths(job,new Path(args[0]));
        FileOutputFormat.setOutputPath(job,new Path(args[1]));

        //将job中配置的相关参数，以及job所用的java类所在的jar包，提交给yarn去运行\

        boolean res = job.waitForCompletion(true);
        System.exit(res?0:1);
    }
}
